from langchain_community.document_compressors.dashscope_rerank import DashScopeRerank


sequence = [
    "人工智能正在深刻地改变着我们的生活。",
    "机器学习是人工智能的一个重要分支。",
    "深度学习在图像识别和自然语言处理中取得了巨大的成功。"
]


reranker = DashScopeRerank(
    model="gte-rerank",
    # other params...
)

# 计算文本相似度
relevance_score_of_top_2 = reranker.rerank(documents=sequence, query="人工智能改变着很多人的生活", top_n=2)
print(relevance_score_of_top_2)